import { Module } from '@nestjs/common';
import { MongooseModule } from '@nestjs/mongoose';
import { GenerationController } from './generation.controller';
import { GenerationService } from './generation.service';
import { GenerationResult, GenerationResultSchema } from './schemas/generation-result.schema';
import { ComfyUIModule } from '../comfyui/comfyui.module';
import { FileUploadModule } from '../common/modules/file-upload.module';
import { WorkflowTemplateModule } from '../workflow-template/workflow-template.module';

@Module({
    imports: [
        MongooseModule.forFeature([
            { name: GenerationResult.name, schema: GenerationResultSchema }
        ]),
        ComfyUIModule,
        FileUploadModule,
        WorkflowTemplateModule
    ],
    controllers: [GenerationController],
    providers: [GenerationService],
    exports: [GenerationService]
})
export class GenerationModule { }